package com.rr;

import com.alibaba.excel.EasyExcel;
import com.alibaba.excel.ExcelWriter;
import com.alibaba.excel.util.ListUtils;
import com.alibaba.excel.write.metadata.WriteSheet;
import com.rr.entity.Student;

import java.util.Date;
import java.util.List;

/**
 * @author zhuangcanrun
 * @Date 2021/11/24/0:39
 * @description
 **/

public class Demo2WriteTest {
    public static void main(String[] args) {
        // 写法1 JDK8+
        // since: 3.0.0-beta1
        String fileName = "easyexcel-demo/src/main/resources/demo2_1.xlsx";
        // 这里 需要指定写用哪个class去写，然后写到第一个sheet，名字为模板 然后文件流会自动关闭
        // 如果这里想使用03 则 传入excelType参数即可
        EasyExcel.write(fileName, Student.class)
                .sheet("模板1")
                .doWrite(() -> {
                    // 分页查询数据
                    return data();
                });

        // 写法2
        fileName = "easyexcel-demo/src/main/resources/demo2_2.xlsx";
        // 这里 需要指定写用哪个class去写，然后写到第一个sheet，名字为模板 然后文件流会自动关闭
        // 如果这里想使用03 则 传入excelType参数即可
        EasyExcel.write(fileName, Student.class).sheet("模板2").doWrite(data());

        // 写法3
        fileName = "easyexcel-demo/src/main/resources/demo2_3.xlsx";
        // 这里 需要指定写用哪个class去写
        ExcelWriter excelWriter = null;
        try {
            excelWriter = EasyExcel.write(fileName, Student.class).build();
            WriteSheet writeSheet = EasyExcel.writerSheet("模板3").build();
            excelWriter.write(data(), writeSheet);
        } finally {
            // 千万别忘记finish 会帮忙关闭流
            if (excelWriter != null) {
                excelWriter.finish();
            }
        }
    }

    private static List<Student> data() {
        List<Student> list = ListUtils.newArrayList();
        for (int i = 0; i < 10; i++) {
            Student data = new Student();
            data.setName("name" + i);
            data.setBirthday(new Date());
            data.setHeight(0.56);
            list.add(data);
        }
        return list;
    }
}
